FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
WORKDIR /train
COPY . .
RUN mv sources.list /etc/apt/sources.list
RUN apt-get update
RUN apt-get install -y build-essential
RUN pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN wandb login 3cfb76268f8ba87e95aaa05c8bcf25c08e90eeab
CMD ["python3","deeplabv3+.py"]
